# -*- coding: utf-8 -*-
# @name             : 数据加密
# @author           : OneYoung
# @create           : 2024/07/18
# @version          : v1.0
# @desc             : 根据需要自行拓展

import hashlib


def get_md5_encryption(string: str, upper=True):
    """
    返回加密结果
    :param string: 加密字符串
    :param upper: 是否返回大写结果，默认True
    """
    md5 = hashlib.md5()
    md5.update(string.encode('utf-8'))
    result = md5.hexdigest()
    md5_str = result.upper() if upper else result
    return md5_str


def get_sha1_encryption(string: str):
    """
    返回sha1加密结果
    :param string: 加密字符串
    """
    md5 = hashlib.sha1()
    md5.update(string.encode('utf-8'))
    return md5.hexdigest()
